PyTorch tensor to numpy dtype